'''
输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，
本题从1开始计数，即链表的尾节点是倒数第1个节点。
例如，一个链表有6个节点，从头节点开始，它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。

给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
'''
class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

def getKthFromEnd(head, k):
    q = []
    cur = head
    while cur:
        q.append(cur)
        cur = cur.next
    # for index,node in enumerate(q):
    #     if len(q)-k == index:
    #         return [q[index].val,q[index].next.val]
    if q[-k].next == None:
         return [q[-k].val]
    # return [q[len(q) - k].val,q[len(q)-k].next.val]#正向取
    return [q[- k].val,q[-k].next.val]#反向取

    #方法2
    # front = head
    # index = 1
    # while True:
    #     if index == k:
    #         break
    #     index += 1
    #     front = front.next
    # while front.next:
    #     head = head.next
    #     front = front.next
    # return head


if __name__ == '__main__':
    l = [2,3,4,5]
    head = ListNode(1)
    cur = head
    for i in l:
        node = ListNode(i)
        cur.next = node
        cur = cur.next
    print(getKthFromEnd(head, 1))
